hackerrank python if-else